import os

MODEL_PROVIDER = os.environ.get("MODEL_PROVIDER", "spark")

from .common import GenerationResult

if MODEL_PROVIDER == "spark":
    from .spark_client import *

elif MODEL_PROVIDER == "openai":
    from .openai_client import *

elif MODEL_PROVIDER == "volc":
    from .volc_client import *

else:
    raise ValueError(f"Unknown MODEL_PROVIDER: {MODEL_PROVIDER}")
